// 简单的向量处理测试脚本
import { saveLancedb, searchLancedb } from './src/core/embed.js';

// 简单的测试数据
const simpleTestData = [
  {
    text: "这是第一个测试文本，关于JavaScript函数的编写和使用。",
    path: "test/file1.js"
  },
  {
    text: "这是第二个测试文本，关于React组件的开发和优化。",
    path: "test/file2.js"
  },
  {
    text: "这是第三个测试文本，关于Node.js服务器的搭建和API设计。",
    path: "test/file3.js"
  }
];

async function testSimpleVector() {
  try {
    console.log('开始简单向量测试...');
    
    // 1. 尝试保存测试数据的向量
    console.log('\n步骤1: 保存测试数据的向量...');
    await saveLancedb(simpleTestData);
    
    // 2. 尝试搜索测试数据
    console.log('\n步骤2: 搜索测试数据...');
    const queries = [
      "JavaScript函数",
      "React组件",
      "Node.js服务器"
    ];
    
    for (const query of queries) {
      console.log(`\n  搜索: "${query}"`);
      const results = await searchLancedb(query, 2);
      
      if (results && results.length > 0) {
        console.log(`  找到 ${results.length} 个结果:`);
        results.forEach((result, index) => {
          console.log(`  结果 ${index + 1}: ${result}`);
        });
      } else {
        console.log('  没有找到结果');
      }
    }
    
  } catch (error) {
    console.error('测试出错:', error);
  }
}

testSimpleVector();